package com.asset.utils;

import com.google.zxing.*;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.common.HybridBinarizer;
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;

import javax.imageio.ImageIO;


import java.awt.*;
import java.awt.geom.RoundRectangle2D;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.OutputStream;
import java.util.Hashtable;

public class QRCodeUtil {
	 private static final String CHARSET = "utf-8";  
     private static final String FORMAT_NAME = "JPG";  
     // 二维码尺寸  
     private static final int QRCODE_SIZE = 200;  
     // LOGO宽度  
     private static final int WIDTH = 60;  
     // LOGO高度  
     private static final int HEIGHT = 50;  
   
     //创建图像缓冲区(即相当于创建一张纸出来)
     private static BufferedImage createImage(String content, String imgPath,  
             boolean needCompress) throws Exception {  
         Hashtable<EncodeHintType, Object> hints = new Hashtable<EncodeHintType, Object>();  
         hints.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.H); //编码容错率 
         hints.put(EncodeHintType.CHARACTER_SET, CHARSET);  //设置编码格式
         hints.put(EncodeHintType.MARGIN, 5);  //设置外围白边宽度
         BitMatrix bitMatrix = new MultiFormatWriter().encode(content,  
                 BarcodeFormat.QR_CODE, QRCODE_SIZE, QRCODE_SIZE, hints);  
         int width = bitMatrix.getWidth();  
         int height = bitMatrix.getHeight();  
         BufferedImage image = new BufferedImage(width, height,  
                 BufferedImage.TYPE_INT_RGB);  
         for (int x = 0; x < width; x++) {  
             for (int y = 0; y < height; y++) {  
                 image.setRGB(x, y, bitMatrix.get(x, y) ? 0xFF000000  
                         : 0xFFFFFFFF);  
             }  
         }  
         if (imgPath == null || "".equals(imgPath)) {  
             return image;  
         }  
         // 插入图片  
         QRCodeUtil.insertImage(image, imgPath, needCompress);  
         return image;  
     }  
   
     /** 
      * 插入LOGO 
      *  
      * @param source 
      *            二维码图片 
      * @param imgPath 
      *            LOGO图片地址 
      * @param needCompress 
      *            是否压缩 
      * @throws Exception 
      */  
     private static void insertImage(BufferedImage source, String imgPath,  
             boolean needCompress) throws Exception {  
         File file = new File(imgPath);  
         if (!file.exists()) {  
             System.err.println(""+imgPath+"   该文件不存在！"); 
             return;  
         }  
         System.out.println("logo地址存在");
         Image src = ImageIO.read(new File(imgPath));  
         int width = src.getWidth(null);  
         int height = src.getHeight(null);  
         if (needCompress) { // 压缩LOGO  
             if (width > WIDTH) {  
                 width = WIDTH;  
             }  
             if (height > HEIGHT) {  
                 height = HEIGHT;  
             }  
             Image image = src.getScaledInstance(width, height,  
                     Image.SCALE_SMOOTH);  
             BufferedImage tag = new BufferedImage(width, height,  
                     BufferedImage.TYPE_INT_RGB);  
             Graphics g = tag.getGraphics();  
             g.drawImage(image, 0, 0, null); // 绘制缩小后的图  
             g.dispose();  
             src = image;  
         }  
         // 插入LOGO  创建一支笔
         Graphics2D graph = source.createGraphics();  
         int x = (QRCODE_SIZE - width) / 2;  
         int y = (QRCODE_SIZE - height) / 2;  
         graph.drawImage(src, x, y, width, height, null);  
         Shape shape = new RoundRectangle2D.Float(x, y, width, width, 6, 6);  
         graph.setStroke(new BasicStroke(3f));  
         graph.draw(shape); 
         graph.dispose();  
     }  
     

    
   
     /** 
      * 生成二维码(内嵌LOGO) 
      *  
      * @param content 
      *            内容 
      * @param imgPath 
      *            LOGO地址 
      * @param destPath 
      *            存放目录 
      * @param needCompress 
      *            是否压缩LOGO 
      * @throws Exception 
      */  
     public static void encode(String content, String imgPath, String destPath,  
             boolean needCompress) throws Exception {  
         BufferedImage image = QRCodeUtil.createImage(content, imgPath,  
                 needCompress);  
         mkdirs(destPath);  
         /*String file = new Random().nextInt(99999999)+".jpg";*/
         String file = "lt.jpg";
         ImageIO.write(image, FORMAT_NAME, new File(destPath+"/"+file));  
     }  
   
     /** 
      * 当文件夹不存在时，mkdirs会自动创建多层目录，区别于mkdir．(mkdir如果父目录不存在则会抛出异常) 
      * @author lanyuan 
      * Email: mmm333zzz520@163.com 
      * @date 2013-12-11 上午10:16:36 
      * @param destPath 存放目录 
      */  
     public static void mkdirs(String destPath) {  
         File file =new File(destPath);      
         //当文件夹不存在时，mkdirs会自动创建多层目录，区别于mkdir．(mkdir如果父目录不存在则会抛出异常)  
         if (!file.exists() && !file.isDirectory()) {  
             file.mkdirs();  
         }  
     }  
   
     /** 
      * 生成二维码(内嵌LOGO) 
      *  
      * @param content 
      *            内容 
      * @param imgPath 
      *            LOGO地址 
      * @param destPath 
      *            存储地址 
      * @throws Exception 
      */  
     //不传压缩参数
     public static void encode(String content, String imgPath, String destPath)  
             throws Exception {  
         QRCodeUtil.encode(content, imgPath, destPath, true);  
     }  
   
     /** 
      * 生成二维码 
      *  
      * @param content 
      *            内容 
      * @param destPath 
      *            存储地址 
      * @param needCompress 
      *            是否压缩LOGO 
      * @throws Exception 
      */  
     //不传logo地址
     public static void encode(String content, String destPath,  
             boolean needCompress) throws Exception {  
         QRCodeUtil.encode(content, null, destPath, needCompress);  
     }  
   
     /** 
      * 生成二维码 
      *  
      * @param content 
      *            内容 
      * @param destPath 
      *            存储地址 
      * @throws Exception 
      */  
     //不传压缩参数和logo地址
     public static void encode(String content, String destPath) throws Exception {  
         QRCodeUtil.encode(content, null, destPath, false);  
     }  
   
     /** 
      * 生成二维码(内嵌LOGO) 
      *  
      * @param content 
      *            内容 
      * @param imgPath 
      *            LOGO地址 
      * @param output 
      *            输出流 
      * @param needCompress 
      *            是否压缩LOGO 
      * @throws Exception 
      */  
     public static void encode(String content, String imgPath,  
             OutputStream output, boolean needCompress) throws Exception {  
         BufferedImage image = QRCodeUtil.createImage(content, imgPath,  
                 needCompress);  
         ImageIO.write(image, FORMAT_NAME, output);  
     }  
   
     /** 
      * 生成二维码 
      *  
      * @param content 
      *            内容 
      * @param output 
      *            输出流 
      * @throws Exception 
      */  
     public static void encode(String content, OutputStream output)  
             throws Exception {  
         QRCodeUtil.encode(content, null, output, false);  
     }  
   
     /** 
      * 解析二维码 
      *  
      * @param file 
      *            二维码图片 
      * @return 
      * @throws Exception 
      */  
     public static String decode(File file) throws Exception {  
         BufferedImage image;  
         image = ImageIO.read(file);  
         if (image == null) {  
             return null;  
         }  
         BufferedImageLuminanceSource source = new BufferedImageLuminanceSource(  
                 image);  
         BinaryBitmap bitmap = new BinaryBitmap(new HybridBinarizer(source));  
         Result result;  
         Hashtable<DecodeHintType, Object> hints = new Hashtable<DecodeHintType, Object>();  
         hints.put(DecodeHintType.CHARACTER_SET, CHARSET);  
         result = new MultiFormatReader().decode(bitmap, hints);  
         String resultStr = result.getText();  
         return resultStr;  
     }  
   
     /** 
      * 解析二维码 
      *  
      * @param path 
      * 二维码图片地址 
      * @return 
      * @throws Exception 
      */  
     public static String decode(String path) throws Exception {  
         return QRCodeUtil.decode(new File(path));  
     }  
   
   
     public static void main(String[] args) throws Exception {  
         String text = "http://www.chinac.com";  
         QRCodeUtil.encode(text, "C:/Users/Administrator/Desktop/chinac.jpg", "C:/Users/Administrator/Desktop",true);
        /* QRCodeUtil.encode(text, "C:/Users/Administrator/Desktop/");*/ 
     } 
}
